"""
需求: 要求使用 UnitTest 框架对数据库查询编写测试脚本
步骤:
    1. 建立数据库连接
    2. 执行查询语句1
    3. 断开数据库连接

    3. 建立数据库连接
    4. 执行查询语句2
    5. 断开数据库连接
"""
import unittest


class TestDB(unittest.TestCase):
    """数据库测试类"""

    def connect(self):
        """建立连接方法"""
        print('建立数据库连接')

    def disconnect(self):
        """断开连接方法"""
        print('断开数据库连接')

    def test_data1(self):
        """查询数据库方法"""
        self.connect()
        print('执行查询语句->基本查询')
        self.disconnect()

    def test_data2(self):
        """查询数据库方法"""
        self.connect()
        print('执行查询语句->连接查询')
        self.disconnect()
